import time
import os

TIME = time.strftime("%m%d_%H%M", time.localtime())

TASK_SUFFIX = ""

DUPLICATE_NUM = 3

TASK_NAME = TIME + TASK_SUFFIX

TASK_DIR = "/home/zhaoxiaoyu/courses/rl/assignments/RL_HW1_Dagger/Dagger/model/play_task/" + TASK_NAME + "/"

IMG_DISPLAY_PATH = TASK_DIR + "display.jpeg"
IMG_DISPLAY_PATH = TASK_DIR + "display.jpeg"

EXPERT_COEF = 0.8

LABEL_PATH = TASK_DIR + "label.txt"

ENABLE_SAVE_IMG = True


def update_taskname():
    global TASK_NAME
    global TASK_DIR
    global IMG_DISPLAY_PATH
    global LABEL_PATH

    TASK_NAME = TIME + TASK_SUFFIX
    TASK_DIR = "/home/zhaoxiaoyu/courses/rl/assignments/RL_HW1_Dagger/Dagger/model/play_task/" + TASK_NAME + "/"
    # TASK_DIR = "./task/"
    if not os.path.exists(TASK_DIR):
        os.makedirs(TASK_DIR)
        os.makedirs(TASK_DIR + "imgs/")
        os.makedirs(TASK_DIR + "expert/")

    IMG_DISPLAY_PATH = TASK_DIR + "display.jpeg"
    LABEL_PATH = TASK_DIR + "label.txt"


if __name__ == "__main__":
    print(TASK_NAME)
    print(TASK_DIR)
    TASK_SUFFIX = "_aSdasd"
    update_taskname()
    print(TASK_NAME)
    print(TASK_DIR)